package com.spzx.product.service.impl;

import com.spzx.common.core.utils.StringUtils;
import com.spzx.product.service.TestService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.core.script.DefaultReactiveScriptExecutor;
import org.springframework.data.redis.core.script.DefaultRedisScript;
import org.springframework.data.redis.core.script.RedisScript;
import org.springframework.stereotype.Service;

import java.util.Arrays;
import java.util.UUID;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.locks.Lock;

@Service
public class TestServiceImpl implements TestService {

    @Autowired
    StringRedisTemplate stringRedisTemplate; //SpringBoot自动化配置

    //    @Override
//    public synchronized void testLock() {
//
//        // 查询Redis中的num值
//         String value = (String) this.stringRedisTemplate.opsForValue().get("num");
//        // 没有该值return
//        if (StringUtils.isBlank(value)) {
//            return;
//        }
//        // 有值就转成成int
//        int num = Integer.parseInt(value);
//        // 把Redis中的num值+1
//        this.stringRedisTemplate.opsForValue().set("num", String.valueOf(++num)); //  java 中  ++ 操作不是原子的。
//
//    }

    //通过redis实现分布式锁  setnx k v 设置成功说明分布式锁成功，设置失败，说明分布式锁被别的线程占用，加锁失败
    @Override
    public void testLock() {

        String uuid = UUID.randomUUID().toString();

        Boolean ifAbsent = stringRedisTemplate.opsForValue().setIfAbsent("lock", "lock",3, TimeUnit.SECONDS);
        if (ifAbsent) {//分布式锁加锁成功

            // 查询Redis中的num值
            String value = (String) this.stringRedisTemplate.opsForValue().get("num");
            // 没有该值return
            if (StringUtils.isBlank(value)) {
                return;
            }
            // 有值就转成成int
            int num = Integer.parseInt(value);
            // 把Redis中的num值+1
            this.stringRedisTemplate.opsForValue().set("num", String.valueOf(++num)); //  java 中  ++ 操作不是原子的。

            //避免释放他人的锁，需要判断锁是否是自己的(判断和删除的操作仍然不是原子性)
//            if (uuid.equals(stringRedisTemplate.opsForValue().get("lock"))) {
//                stringRedisTemplate.delete("lock");
//            }
            //采用LUA脚本来保证删除原子性

            String script = "if redis.call(\"get\",KEYS[1]) == ARGV[1]\n" +
                    "then\n" +
                    "    return redis.call(\"del\",KEYS[1])\n" +
                    "else\n" +
                    "    return 0\n" +
                    "end";
            DefaultRedisScript<Long> redisScript = new DefaultRedisScript<>();
            redisScript.setScriptText(script);
            redisScript.setResultType(Long.class);
            Long result = stringRedisTemplate.execute(redisScript, Arrays.asList("lock"), uuid);
            System.out.println("result = " + result);

        } else {
            try {
                Thread.sleep(200);
            } catch (InterruptedException e) {
                throw new RuntimeException(e);
            }
            this.testLock();//递归 自旋抢锁
        }

    }
}
